package com.lenl.datastruct.hashtable;

import java.util.Scanner;

/**
 * @author Lenl
 * @version v1.0
 * @create 2022-05-02 20:26
 * @description 哈希表
 */
public class HashTableDemo {
    public static void main(String[] args) {
        //创建一个哈希表
        HashTable hashTable=new HashTable(7);
        //写一个简单菜单
        String key="";
        while (true){
            Scanner scanner=new Scanner(System.in);
            System.out.println("add:添加雇员");
            System.out.println("list:显示雇员");
            System.out.println("find:查找雇员");
            System.out.println("exit:退出系统");
            key=scanner.next();
            switch(key){
                case "add":
                    System.out.println("输入id:");
                    int id=scanner.nextInt();
                    System.out.println("输入名字：");
                    String name=scanner.next();
                    //创建雇员
                    Emp emp=new Emp(id,name);
                    hashTable.add(emp);
                    break;
                case "list":
                    hashTable.list();
                    break;
                case "find":
                    System.out.println("输入查找的id:");
                    id=scanner.nextInt();
                    hashTable.findById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.out.println("退出系统");
                    System.exit(0);
                default:
                    break;
            }
        }
    }
}
//创建HashTable
class HashTable{
    private EmpLinkedList[] empLinkedListArray;
    private int size;
    public  HashTable(int size){
        //初始化
        this.size=size;
        empLinkedListArray=new EmpLinkedList[size];
        //？一个坑 要分别初始化每条列表
        for (int i=0;i<size;i++){
            empLinkedListArray[i]=new EmpLinkedList();
        }
    }
    //添加雇员
    public void add(Emp emp){
        //根据员工id得到该员工应该加入的链表
        int empLinkedListNo=hashFun(emp.id);
        //将emp添加到对应的链表中
        empLinkedListArray[empLinkedListNo].add(emp);

    }
    //遍历所有的链表:遍历哈希表
    public void list(){
        for (int i=0;i<size;i++){
            empLinkedListArray[i].list(i);
        }
    }

    //根据id找雇员
    public void findById(int id){
        int empLinkedNo=hashFun(id);
        Emp emp=empLinkedListArray[empLinkedNo].findById(id);
        if(emp!=null){
            //找到
            System.out.println("在第"+empLinkedNo+"条链表找到雇员"+emp.name);
        } else{
            System.out.println("未找到该雇员");
        }
    }


    //编写散列函数，使用一个简单的取模法
    public int hashFun(int id){
        return id%size;
    }

}



//表示一个雇员
class Emp{
    public int id;
    public String name;
    public Emp next; //默认为null
    public Emp(int id,String name){
        super();
        this.id=id;
        this.name=name;
    }
}

//创建一个EmpLinkedList,表示链表
class EmpLinkedList{
    //头指针，指向第一个Emp，因此我们这个链表的head直接指向第一个Emp
    private Emp head;//默认null
    //添加雇员到链表
    public void add(Emp emp){
        //假定添加雇员时，id自增长的，即id分配总是从小到大
        //因此将该雇员直接加入到本链条最后即可
        //如果添加第一个雇员
        if(head==null){
            head=emp;
            return;
        }
        //不是第一个雇员，则使用辅助指针定位到最后
        Emp curEmp=head;
        while(true){
            if(curEmp.next==null)
                break;
            curEmp=curEmp.next;
        }
        //加入链表
        curEmp.next=emp;
    }
    //遍历雇员信息
    public void list(int no){
        if(head==null){
            System.out.println("第"+(no+1)+"条链表为空");
            return;
        }
        System.out.print("第"+(no+1)+"条链表信息为：");
        Emp emp=head;//辅助指针
        while(true){
            System.out.printf("=>id=%d,name=%s\t",emp.id,emp.name);
            if(emp.next==null){
                //最后一个节点
                break;
            }
            emp=emp.next;
        }
        System.out.println();
    }
    //根据id找
    public Emp findById(int id){
        if(head==null){
            System.out.println("链表为空");
            return null;
        }
        //辅助指针
        Emp curEmp=head;
        while(true){
            if(curEmp.id==id){
                break;
            }
            if(curEmp.next==null){
                //没有找到
                curEmp=null;
                break;
            }
            curEmp=curEmp.next;
        }
        return curEmp;
    }

}